import torch
import torch.nn as nn



class Roc(nn.Module):
    def __init__(self):
        super().__init__()
        self.conv1 = nn.Conv2d(1, 20, 5)
        self.conv2 = nn.Conv2d(20, 20, 5)

    def forward(self, x):
        y = x + 1
        return y


roc = Roc()
x = torch.tensor(1.0)
y = roc(x)
print(y)

